import openai

API_KEY = "sk-ioSk7SbGiIu40lnOINybFdlyf2oOMbteB1dQ4jRwrhpqONtQ"
BASE_URL = "https://api.getaiai.com"
MODEL_NAME = "gpt-4o"

client = openai.Client(api_key="sk-ioSk7SbGiIu40lnOINybFdlyf2oOMbteB1dQ4jRwrhpqONtQ", base_url="https://api.getaiai.com/v1")
EXAMPLE_COMPANY = """
公司规章制度

第一条：工作时间

1.全体员工每日工作时间为10小时，每周工作6天，无需固定休息日。

第二条：计件工作管理

1.对计件工作的员工，公司自行决定劳动定额，不受每日8小时工作时间限制，报酬以实际产出为准。

第三条：节假日安排

1.公司仅在春节放假3天，其他节假日正常上班，无需安排员工休假。

第四条：特殊工作制度

1.因公司业务需求，所有部门实行全年无休制度，休息时间需员工自行协调，每人每月最多休假1天。

"""

EXAMPLE_LAW = """
第三十六条　国家实行劳动者每日工作时间不超过八小时、平均每周工作时间不超过四十四小时的工时制度。

第三十七条　对实行计件工作的劳动者，用人单位应当根据本法第三十六条规定的工时制度合理确定其劳动定额和计件报酬标准。

第三十八条　用人单位应当保证劳动者每周至少休息一日。

第三十九条　企业因生产特点不能实行本法第三十六条、第三十八条规定的，经劳动行政部门批准，可以实行其他工作和休息办法。

第四十条　用人单位在下列节日期间应当依法安排劳动者休假：

（一）元旦；

（二）春节；

（三）国际劳动节；

（四）国庆节；

（五）法律、法规规定的其他休假节日。
"""
SYSTEM_PROMPT = """You will behave as an expert legal advisor, proficient in Chinese law. 
You have a comprehensive understanding of the legal system in China, with extensive academic qualifications, including a PhD in law. 
Your primary role is to provide expert legal advice, drawing on your deep knowledge of Chinese legislation, regulations, and legal precedents. 
You can proficiently quote Chinese laws to support your arguments and provide relevant real-life examples from within China for both prosecution and defense scenarios. 
It is essential that your legal advice be professional and accurate, as you play a crucial role in informing and guiding users about legal matters within the Chinese context. 
You must communicate exclusively in Chinese, ensuring that your responses are tailored to the linguistic and cultural nuances of the Chinese legal system.

你将使用中文回答用户的问题，除非用户使用的是其他语言。
下面用户将提供相关法律法规片段和公司的文件，你将参考用户提供的法律法规片段和其他任何可能相关的中国法律，检查这些文件并返回不符合法规的内容
你将回复JSON格式字符串，JSON定义如下：
{ "result" : [{"content":"<公司文件片段>", "law":"违反《中华人民共和国XXX法》第<>条、第<>条"}]}
result: 检查的结果list
content: 不符合法规的内容部分，尽可能精简
law: 违反的法律法规，尽可能精简
你将输出完整的plain json字符串，不包括markdown格式等其他内容。
"""

USER_MESSAGE = f"法律片段如下：\n{EXAMPLE_LAW}\n\n\n公司文件如下：\n{EXAMPLE_COMPANY}"


def main():
    completion = client.chat.completions.create(
        model=MODEL_NAME,
        messages=[
            {"role": "system", "content": SYSTEM_PROMPT},
            {"role": "user", "content": USER_MESSAGE},
        ],
        response_format={"type": "json_object"},
    )
    print(completion.choices[0].message.content)


if __name__ == '__main__':
    main()
